/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package com.jfoenix.transitions;

import javafx.scene.CacheHint;
import javafx.scene.Node;
import javafx.scene.layout.Region;

import java.util.HashMap;

public class CacheMomento {
    private static HashMap<Node, CacheMomento> history = new HashMap<>();

    private boolean cache;
    private boolean cacheShape;
    private boolean snapToPixel;
    private CacheHint cacheHint = CacheHint.DEFAULT;
    private Node node;

    public CacheMomento(Node node) {
        this.node = node;
    }

    /**
     * this method will cache the node only if it wasn't cached before
     */
    public void cache() {
        if (!history.containsKey(node)) {
            this.cache = node.isCache();
            this.cacheHint = node.getCacheHint();
            node.setCache(true);
            node.setCacheHint(CacheHint.SPEED);
            if (node instanceof Region) {
                this.cacheShape = ((Region) node).isCacheShape();
                this.snapToPixel = ((Region) node).isSnapToPixel();
                ((Region) node).setCacheShape(true);
                ((Region) node).setSnapToPixel(true);
            }
            history.put(node, this);
        } else {
            CacheMomento cached = new CacheMomento(node);
            this.cache = cached.cache;
            this.cacheHint = cached.cacheHint;
            this.cacheShape = cached.cacheShape;
            this.snapToPixel = cached.snapToPixel;
        }

    }

    public void restore() {
        node.setCache(cache);
        node.setCacheHint(cacheHint);
        if (node instanceof Region) {
            ((Region) node).setCacheShape(cacheShape);
            ((Region) node).setSnapToPixel(snapToPixel);
        }
        history.remove(node);
    }
}
